package com.liqi.web.controller.system;

import java.util.List;
import java.util.Set;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import com.liqi.common.constant.Constants;
import com.liqi.common.core.domain.AjaxResult;
import com.liqi.common.core.domain.entity.SysMenu;
import com.liqi.common.core.domain.entity.SysUser;
import com.liqi.common.core.domain.model.LoginBody;
import com.liqi.common.utils.SecurityUtils;
import com.liqi.common.utils.encrypt.AesEncryption;
import com.liqi.framework.web.service.SysLoginService;
import com.liqi.framework.web.service.SysPermissionService;
import com.liqi.system.service.ISysMenuService;

/**
 * 登录验证
 * 
 * @author ruoyi
 */
@RestController
public class SysLoginController {
	private static final Logger log = LoggerFactory.getLogger(SysLoginController.class);
	
	@Autowired
	private SysLoginService loginService;

	@Autowired
	private ISysMenuService menuService;

	@Autowired
	private SysPermissionService permissionService;

	/**
	 * 登录方法
	 * 
	 * @param loginBody 登录信息
	 * @return 结果
	 */
	@PostMapping("/login")
	public AjaxResult login(@RequestBody LoginBody loginBody) throws Exception {
		AjaxResult ajax = AjaxResult.success();
		// 生成令牌
		String token = loginService.login(loginBody.getUsername(),
				AesEncryption.desEncryptForJs(loginBody.getPassword()), loginBody.getCode(), loginBody.getUuid());
		ajax.put(Constants.TOKEN, token);
		return ajax;
	}

	/**
	 * 登录方法
	 * 
	 * @param loginBody 登录信息
	 * @return 结果
	 */
	@GetMapping("/loginByToken")
	public void loginByToken(HttpServletRequest request, HttpServletResponse response,
			@RequestParam(value = "token", required = true) String token) throws Exception {
		long time = System.currentTimeMillis();
		AjaxResult ajax = AjaxResult.success();
		// 生成令牌
		String systemToken = loginService.loginByToken(token);
		ajax.put(Constants.TOKEN, systemToken);

		long time2 = System.currentTimeMillis();
		
		log.info("loginByToken单点登录，时间差：" + (time2 - time) + " ms");
		
		Cookie cookie = new Cookie("Admin-Token", systemToken);
		cookie.setMaxAge(1800);
		cookie.setPath("/");
		cookie.setHttpOnly(true);
		response.addCookie(cookie);
		response.sendRedirect("/SelfServiceManagementSite/index");

		// return ajax;
	}

	/**
	 * 获取用户信息
	 * 
	 * @return 用户信息
	 */
	@GetMapping("getInfo")
	public AjaxResult getInfo() {
		SysUser user = SecurityUtils.getLoginUser().getUser();
		// 角色集合
		Set<String> roles = permissionService.getRolePermission(user);
		// 权限集合
		Set<String> permissions = permissionService.getMenuPermission(user);
		AjaxResult ajax = AjaxResult.success();
		ajax.put("user", user);
		ajax.put("roles", roles);
		ajax.put("permissions", permissions);
		return ajax;
	}

	/**
	 * 获取路由信息
	 * 
	 * @return 路由信息
	 */
	@GetMapping("getRouters")
	public AjaxResult getRouters() {
		Long userId = SecurityUtils.getUserId();
		List<SysMenu> menus = menuService.selectMenuTreeByUserId(userId);
		return AjaxResult.success(menuService.buildMenus(menus));
	}
}
